# coding:utf-8
#import self as self

from lib.base import Base
from page.android import flight_list_page, train_list_page


class FlightListToTrain(Base):
    def __init__(self, method_name):
        Base.__init__(self, methodName=method_name)

        """
        本页元素
        """
        self.jiangjia_btn = flight_list_page.elements['降价提醒']
        self.trainline = flight_list_page.elements['火车路线']

        """
        外部页面元素
        """
        self.train_btn = train_list_page.elements['火车按钮']

    def test_flight_to_train(self):
        self.open_schema_android(6019)
        self.check(self.jiangjia_btn)
        self.click(self.trainline)
        self.check(self.train_btn)
